from loss.loss import MSELoss, SmoothL1, WingLoss, PFLDLoss


LOSS = {"mse": MSELoss,
        "smoothl1": SmoothL1,
        "wing": WingLoss,
        "PFLDLoss":PFLDLoss}